import os
from dotenv import load_dotenv

# 加载环境变量
load_dotenv()

# 本地模型配置
DEFAULT_MODEL = "runwayml/stable-diffusion-v1-5"
MODEL_CACHE_DIR = "model_cache"

# 提示词矩阵配置
PROMPT_MATRIX_CATEGORIES = {
    "tone": ["正式", "随意", "幽默", "专业", "简洁"],
    "detail_level": ["简略", "中等", "详细", "非常详细"],
    "format": ["段落", "列表", "表格", "要点"],
    "perspective": ["用户视角", "专家视角", "第三方视角"]
}

# 多模态处理配置
SUPPORTED_IMAGE_FORMATS = ['.jpg', '.jpeg', '.png', '.gif']
MAX_IMAGE_SIZE = (1024, 1024)

# 输出配置
OUTPUT_DIR = "outputs"
EXAMPLES_DIR = "examples"

# 创建输出目录
os.makedirs(OUTPUT_DIR, exist_ok=True)
os.makedirs(EXAMPLES_DIR, exist_ok=True)
os.makedirs(MODEL_CACHE_DIR, exist_ok=True)